import os

def check_photo_prefixes(input_folder):
    # 遍历输入文件夹下的所有子文件夹
    for root, dirs, files in os.walk(input_folder):
        for dir in dirs:
            # 获取子文件夹中的所有文件
            sub_items = os.listdir(os.path.join(root, dir))
            all_prefix_same = True
            for sub_item in sub_items:
                # 如果是照片（以.jpg, .jpeg或.png结尾）
                if sub_item.endswith('.jpg') or sub_item.endswith('.jpeg') or sub_item.endswith('.png'):
                    # 检查照片的文件名是否以子文件夹名开头
                    if not sub_item.startswith(dir):
                        all_prefix_same = False
                        break
            if all_prefix_same:
                print(f'文件夹 {dir} 中，所有照片的前缀都与文件夹名相同。')
            else:
                print(f'文件夹 {dir} 中，存在照片的前缀与文件夹名不同。')

# 调用函数，指定输入文件夹
check_photo_prefixes('traindata_gt_guiji_old')
